/*
 * Copyright (c) 2011-2018, Meituan Dianping. All Rights Reserved.
 *
 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements. See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License. You may obtain a copy of the License at
 *
 *    http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */
package com.dianping.cat.report.page.storage.transform;

import com.dianping.cat.Constants;
import com.dianping.cat.consumer.storage.model.entity.StorageReport;

public class StorageMergeHelper {

	public StorageReport mergeAllDomains(StorageReport storageReport, String domain) {
		if (Constants.ALL.equals(domain)) {
			AllDomainMerger merger = new AllDomainMerger();

			merger.visitStorageReport(storageReport);
			storageReport = merger.getStorageReport();
		}
		return storageReport;
	}

	public StorageReport mergeAllMachines(StorageReport storageReport, String ipAddress) {
		if (Constants.ALL.equals(ipAddress)) {
			AllMachineMerger merger = new AllMachineMerger();

			merger.visitStorageReport(storageReport);
			storageReport = merger.getStorageReport();
		}
		return storageReport;
	}

	public StorageReport mergeReport(StorageReport report, String ipAddress, String domain) {
		StorageReport temp = mergeAllMachines(report, ipAddress);

		return mergeAllDomains(temp, domain);
	}
}
